home *** CD-ROM | disk | FTP | other *** search
/ Mac-Source 1994 July / Mac-Source_July_1994.iso / C and C++ / Miscellaneous / gdbm / gdbmopen.c < prev    next >
Text File  |  1992-04-05  |  12KB  |  401 lines
  1. /* gdbmopen.c - Open the dbm file and initialize data structures for use. */
  2.  
  3. /*  This file is part of GDBM, the GNU data base manager, by Philip A. Nelson.
  4.     Copyright (C) 1990, 1991  Free Software Foundation, Inc.
  5.  
  6.     GDBM is free software; you can redistribute it and/or modify
  7.     it under the terms of the GNU General Public License as published by
  8.     the Free Software Foundation; either version 1, or (at your option)
  9.     any later version.
  10.  
  11.     GDBM is distributed in the hope that it will be useful,
  12.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.     GNU General Public License for more details.
  15.  
  16.     You should have received a copy of the GNU General Public License
  17.     along with GDBM; see the file COPYING.  If not, write to
  18.     the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  19.  
  20.     You may contact the author by:
  21.        e-mail:  phil@cs.wwu.edu
  22.       us-mail:  Philip A. Nelson
  23.                 Computer Science Department
  24.                 Western Washington University
  25.                 Bellingham, WA 98226
  26.         phone:  (206) 676-3035
  27.        
  28. *************************************************************************/
  29.  
  30.  
  31. #include "gdbmdefs.h"
  32. #include "gdbmerrno.h"
  33.  
  34. extern gdbm_error gdbm_errno;
  35.  
  36.  
  37.  
  38. /* Initialize dbm system.  FILE is a pointer to the file name.  If the file
  39.    has a size of zero bytes, a file initialization procedure is performed,
  40.    setting up the initial structure in the file.  BLOCK_SIZE is used during
  41.    initialization to determine the size of various constructs.  If the value
  42.    is less than 512, the file system blocksize is used, otherwise the value
  43.    of BLOCK_SIZE is used.  BLOCK_SIZE is ignored if the file has previously
  44.    initialized.  If READ_WRITE is set to GDBM_READ the user wants to just
  45.    read the database and any call to dbm_store or dbm_delete will fail. Many
  46.    readers can access the database at the same time.  If READ_WRITE is set to
  47.    GDBM_WRITE, the user wants both read and write access to the database and
  48.    requires exclusive access.  If READ_WRITE is GDBM_WRCREAT, the user wants
  49.    both read and write access to the database and if the database does not
  50.    exist, create a new one.  If READ_WRITE is GDBM_NEWDB, the user want a
  51.    new database created, regardless of whether one existed, and wants read
  52.    and write access to the new database.  Any error detected will cause a 
  53.    return value of null and an approprate value will be in gdbm_errno.  If
  54.    no errors occur, a pointer to the "gdbm file descriptor" will be
  55.    returned. */
  56.    
  57.  
  58. gdbm_file_info *
  59. gdbm_open (file, block_size, read_write, mode, fatal_func)
  60.      char *file;
  61.      int  block_size;
  62.      int  read_write;
  63.      int  mode;
  64.      void (*fatal_func) ();
  65. {
  66. #ifdef macintosh
  67. #pragma unused(mode)
  68. #endif
  69.   gdbm_file_info *dbf;        /* The record to return. */
  70.   struct stat file_stat;    /* Space for the stat information. */
  71.   int         len;        /* Length of the file name. */
  72.   int         num_bytes;    /* Used in reading and writing. */
  73.   long        file_pos;        /* Used with seeks. */
  74.   int          lock_val;         /* Returned by the flock call. */
  75.   int          file_block_size;    /* Block size to use for a new file. */
  76.   int           index;        /* Used as a loop index. */
  77.   char        need_trunc;    /* Used with GDBM_NEWDB and locking to avoid
  78.                    truncating a file from under a reader. */
  79.  
  80.   /* Allocate new info structure. */
  81.   dbf = (gdbm_file_info *) malloc (sizeof (gdbm_file_info));
  82.   if (dbf == NULL)
  83.     {
  84.       gdbm_errno = GDBM_MALLOC_ERROR;
  85.       return NULL;
  86.     }
  87.  
  88.   /* Initialize some fields for known values.  This is done so gdbm_close
  89.      will work if called before allocating some structures. */
  90.   dbf->dir  = NULL;
  91.   dbf->bucket = NULL;
  92.   for (index = 0; index < CACHE_SIZE; index++)
  93.     {
  94.       dbf->bucket_cache[index].ca_bucket = NULL;
  95.       dbf->bucket_cache[index].ca_data.dptr = NULL;
  96.     }
  97.   
  98.   /* Save name of file. */
  99.   len = strlen (file);
  100.   dbf->name = (char *) malloc (len + 1);
  101.   if (dbf->name == NULL)
  102.     {
  103.       free (dbf);
  104.       gdbm_errno = GDBM_MALLOC_ERROR;
  105.       return NULL;
  106.     }
  107.   strcpy (dbf->name, file);
  108.  
  109.   /* Initialize the fatal error routine. */
  110.   dbf->fatal_err = fatal_func;
  111.  
  112.   /* Open the file. */
  113.   need_trunc = FALSE;
  114.   if (read_write == GDBM_READER)
  115.     {
  116.       dbf->desc = OPEN(dbf->name, O_RDONLY, 0);
  117.     }
  118.   else if (read_write == GDBM_WRITER)
  119.     {
  120.       dbf->desc = OPEN(dbf->name, O_RDWR, 0);
  121.     }
  122.   else if (read_write == GDBM_NEWDB)
  123.     {
  124.       dbf->desc = OPEN(dbf->name, O_RDWR|O_CREAT, mode);
  125.       read_write = GDBM_WRITER;
  126.       need_trunc = TRUE;
  127.     }
  128.   else
  129.     {
  130.       dbf->desc = OPEN(dbf->name, O_RDWR|O_CREAT, mode);
  131.       read_write = GDBM_WRITER;
  132.     }
  133.   if (dbf->desc < 0)
  134.     {
  135.       free (dbf->name);
  136.       free (dbf);
  137.       gdbm_errno = GDBM_FILE_OPEN_ERROR;
  138.       return NULL;
  139.     }
  140.  
  141.   /* Get the status of the file. */
  142.   fstat (dbf->desc, &file_stat);
  143.  
  144.   /* Lock the file in the approprate way. */
  145.   if (read_write == GDBM_READER)
  146.     {
  147.       if (file_stat.st_size == 0)
  148.     {
  149.       close (dbf->desc);
  150.       free (dbf->name);
  151.       free (dbf);
  152.       gdbm_errno = GDBM_EMPTY_DATABASE;
  153.       return NULL;
  154.     }
  155.       /* Sets lock_val to 0 for success.  See systems.h. */
  156.       READLOCK_FILE(dbf);
  157.     }
  158.   else
  159.     {
  160.       /* Sets lock_val to 0 for success.  See systems.h. */
  161.       WRITELOCK_FILE(dbf);
  162.     }
  163.   if (lock_val != 0)
  164.     {
  165.       close (dbf->desc);
  166.       free (dbf->name);
  167.       free (dbf);
  168.       if (read_write == GDBM_READER)
  169.     gdbm_errno = GDBM_CANT_BE_READER;
  170.       else
  171.     gdbm_errno = GDBM_CANT_BE_WRITER;
  172.       return NULL;
  173.     }
  174.  
  175.   /* Record the kind of user. */
  176.   dbf->read_write = read_write;
  177.  
  178.   /* If we do have a write lock and it was a GDBM_NEWDB, it is 
  179.      now time to truncate the file. */
  180.   if (need_trunc && file_stat.st_size != 0)
  181.     {
  182.       TRUNCATE (dbf);
  183.       fstat (dbf->desc, &file_stat);
  184.     }
  185.  
  186.   /* Decide if this is a new file or an old file. */
  187.   if (file_stat.st_size == 0)
  188.     {
  189.  
  190.       /* This is a new file.  Create an empty database.  */
  191.  
  192.       /* Start with the blocksize. */
  193.       if (block_size < 512)
  194.     file_block_size = STATBLKSIZE;
  195.       else
  196.     file_block_size = block_size;
  197.  
  198.       /* Get space for the file header. */
  199.       dbf->header = (gdbm_file_header *) malloc (file_block_size);
  200.       if (dbf->header == NULL)
  201.     {
  202.       gdbm_close (dbf);
  203.       gdbm_errno = GDBM_MALLOC_ERROR;
  204.       return NULL;
  205.     }
  206.  
  207.       /* Set the magic number and the block_size. */
  208.       dbf->header->header_magic = 0x13579ace;
  209.       dbf->header->block_size = file_block_size;
  210.      
  211.       /* Create the initial hash table directory.  */
  212.       dbf->header->dir_size = 8 * sizeof (int);
  213.       dbf->header->dir_bits = 3;
  214.       while (dbf->header->dir_size < dbf->header->block_size)
  215.     {
  216.       dbf->header->dir_size <<= 1;
  217.       dbf->header->dir_bits += 1;
  218.     }
  219.  
  220.       /* Check for correct block_size. */
  221.       if (dbf->header->dir_size != dbf->header->block_size)
  222.     {
  223.       gdbm_close (dbf);
  224.       gdbm_errno = GDBM_BLOCK_SIZE_ERROR;
  225.       return NULL;
  226.     }
  227.  
  228.       /* Allocate the space for the directory. */
  229.       dbf->dir = (int *) malloc (dbf->header->dir_size);
  230.       if (dbf->dir == NULL)
  231.     {
  232.       gdbm_close (dbf);
  233.       gdbm_errno = GDBM_MALLOC_ERROR;
  234.       return NULL;
  235.     }
  236.       dbf->header->dir = dbf->header->block_size;
  237.  
  238.       /* Create the first and only hash bucket. */
  239.       dbf->header->bucket_elems =
  240.     (dbf->header->block_size - sizeof (hash_bucket))
  241.     / sizeof (bucket_element) + 1;
  242.       dbf->header->bucket_size  = dbf->header->block_size;
  243.       dbf->bucket = (hash_bucket *) (alloca (dbf->header->bucket_size));
  244.       if (dbf->bucket == NULL)
  245.     {
  246.       gdbm_close (dbf);
  247.       gdbm_errno = GDBM_MALLOC_ERROR;
  248.       return NULL;
  249.     }
  250.       _gdbm_new_bucket (dbf, dbf->bucket, 0);
  251.       dbf->bucket->av_count = 1;
  252.       dbf->bucket->bucket_avail[0].av_adr = 3*dbf->header->block_size;
  253.       dbf->bucket->bucket_avail[0].av_size = dbf->header->block_size;
  254.  
  255.       /* Set table entries to point to hash buckets. */
  256.       for (index = 0; index < dbf->header->dir_size / sizeof (int); index++)
  257.     dbf->dir[index] = 2*dbf->header->block_size;
  258.  
  259.       /* Initialize the active avail block. */
  260.       dbf->header->avail.size
  261.     = ( (dbf->header->block_size - sizeof (gdbm_file_header))
  262.      / sizeof (avail_elem)) + 1;
  263.       dbf->header->avail.count = 0;
  264.       dbf->header->avail.next_block = 0;
  265.       dbf->header->next_block  = 4*dbf->header->block_size;
  266.  
  267.       /* Write initial configuration to the file. */
  268.       /* Block 0 is the file header and active avail block. */
  269.       num_bytes = write (dbf->desc, (char *) dbf->header, dbf->header->block_size);
  270.       if (num_bytes != dbf->header->block_size)
  271.     {
  272.       gdbm_close (dbf);
  273.       gdbm_errno = GDBM_FILE_WRITE_ERROR;
  274.       return NULL;
  275.     }
  276.  
  277.       /* Block 1 is the initial bucket directory. */
  278.       num_bytes = write (dbf->desc, (char *) dbf->dir, dbf->header->dir_size);
  279.       if (num_bytes != dbf->header->dir_size)
  280.     {
  281.       gdbm_close (dbf);
  282.       gdbm_errno = GDBM_FILE_WRITE_ERROR;
  283.       return NULL;
  284.     }
  285.  
  286.       /* Block 2 is the only bucket. */
  287.       num_bytes = write (dbf->desc, (char *) dbf->bucket, dbf->header->bucket_size);
  288.       if (num_bytes != dbf->header->bucket_size)
  289.     {
  290.       gdbm_close (dbf);
  291.       gdbm_errno = GDBM_FILE_WRITE_ERROR;
  292.       return NULL;
  293.     }
  294.  
  295.       /* Wait for initial configuration to be written to disk. */
  296.       fsync (dbf->desc);
  297.  
  298.     }
  299.   else
  300.     {
  301.       /* This is an old database.  Read in the information from the file
  302.      header and initialize the hash directory. */
  303.  
  304.       gdbm_file_header partial_header;  /* For the first part of it. */
  305.  
  306.       /* Read the partial file header. */
  307.       num_bytes = read (dbf->desc, (char *) &partial_header, sizeof (gdbm_file_header));
  308.       if (num_bytes != sizeof (gdbm_file_header))
  309.     {
  310.       gdbm_close (dbf);
  311.       gdbm_errno = GDBM_FILE_READ_ERROR;
  312.       return NULL;
  313.     }
  314.  
  315.       /* Is the magic number good? */
  316.       if (partial_header.header_magic != 0x13579ace)
  317.     {
  318.       gdbm_close (dbf);
  319.       gdbm_errno = GDBM_BAD_MAGIC_NUMBER;
  320.       return NULL;
  321.     }
  322.  
  323.       /* It is a good database, read the entire header. */
  324.       dbf->header = (gdbm_file_header *) malloc (partial_header.block_size);
  325.       if (dbf->header == NULL)
  326.     {
  327.       gdbm_close (dbf);
  328.       gdbm_errno = GDBM_MALLOC_ERROR;
  329.       return NULL;
  330.     }
  331.       bcopy (&partial_header, dbf->header, sizeof (gdbm_file_header));
  332.       num_bytes = read (dbf->desc, (char *) &dbf->header->avail.av_table[1],
  333.             dbf->header->block_size-sizeof (gdbm_file_header));
  334.       if (num_bytes != dbf->header->block_size-sizeof (gdbm_file_header))
  335.     {
  336.       gdbm_close (dbf);
  337.       gdbm_errno = GDBM_FILE_READ_ERROR;
  338.       return NULL;
  339.     }
  340.     
  341.       /* Allocate space for the hash table directory.  */
  342.       dbf->dir = (int *) malloc (dbf->header->dir_size);
  343.       if (dbf->dir == NULL)
  344.     {
  345.       gdbm_close (dbf);
  346.       gdbm_errno = GDBM_MALLOC_ERROR;
  347.       return NULL;
  348.     }
  349.  
  350.       /* Read the hash table directory. */
  351.       file_pos = lseek (dbf->desc, dbf->header->dir, L_SET);
  352.       if (file_pos != dbf->header->dir)
  353.     {
  354.       gdbm_close (dbf);
  355.       gdbm_errno = GDBM_FILE_SEEK_ERROR;
  356.       return NULL;
  357.     }
  358.  
  359.       num_bytes = read (dbf->desc, (char *) dbf->dir, dbf->header->dir_size);
  360.       if (num_bytes != dbf->header->dir_size)
  361.     {
  362.       gdbm_close (dbf);
  363.       gdbm_errno = GDBM_FILE_READ_ERROR;
  364.       return NULL;
  365.     }
  366.  
  367.     }
  368.       
  369.   /* Initialize the bucket cache. */
  370.   for (index = 0; index < CACHE_SIZE; index++)
  371.     {
  372.       dbf->bucket_cache[index].ca_bucket
  373.     = (hash_bucket *) malloc (dbf->header->bucket_size);
  374.       if (dbf->bucket_cache[index].ca_bucket == NULL)
  375.     {
  376.       gdbm_close (dbf);
  377.       gdbm_errno = GDBM_MALLOC_ERROR;
  378.       return NULL;
  379.     }
  380.       dbf->bucket_cache[index].ca_adr = 0;
  381.       dbf->bucket_cache[index].ca_changed = FALSE;
  382.       dbf->bucket_cache[index].ca_data.hash_val = -1;
  383.       dbf->bucket_cache[index].ca_data.elem_loc = -1;
  384.     }
  385.  
  386.   /* Finish initializing dbf. */
  387.   dbf->last_read = -1;
  388.   dbf->bucket = dbf->bucket_cache[0].ca_bucket;
  389.   dbf->bucket_dir = 0;
  390.   dbf->cache_entry = &dbf->bucket_cache[0];
  391.   dbf->header_changed = FALSE;
  392.   dbf->directory_changed = FALSE;
  393.   dbf->bucket_changed = FALSE;
  394.   dbf->second_changed = FALSE;
  395.   
  396.   /* Everything is fine, return the pointer to the file
  397.      information structure.  */
  398.   return dbf;
  399.  
  400. }
  401.